#include "DXUT.h"
#include <windows.h>
#include <stdio.h>
#include <fcntl.h>
#include <io.h>
#include <iostream>
#include <fstream>
#ifndef _USE_OLD_IOSTREAMS
using namespace std;
#endif
// maximum mumber of lines the output console should have
static const WORD MAX_CONSOLE_LINES = 1024;
//#ifdef _DEBUG

void RedirectIOToConsole()
{
	int hConHandle;
	long lStdHandle;
	CONSOLE_SCREEN_BUFFER_INFO coninfo;
	FILE *fp;
	// allocate a console for this app
	AllocConsole();
	// set the screen buffer to be big enough to let us scroll text
	GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE),&coninfo);
	coninfo.dwSize.Y = MAX_CONSOLE_LINES;
	SetConsoleScreenBufferSize(GetStdHandle(STD_OUTPUT_HANDLE),coninfo.dwSize);
	// redirect unbuffered STDOUT to the console
	lStdHandle = (long)GetStdHandle(STD_OUTPUT_HANDLE);
	hConHandle = _open_osfhandle(lStdHandle, _O_TEXT);
	fp = _fdopen( hConHandle, "w" );
	*stdout = *fp;
	setvbuf( stdout, NULL, _IONBF, 0 );
	// redirect unbuffered STDIN to the console
	lStdHandle = (long)GetStdHandle(STD_INPUT_HANDLE);
	hConHandle = _open_osfhandle(lStdHandle, _O_TEXT);
	fp = _fdopen( hConHandle, "r" );
	*stdin = *fp;
	setvbuf( stdin, NULL, _IONBF, 0 );
	// redirect unbuffered STDERR to the console
	lStdHandle = (long)GetStdHandle(STD_ERROR_HANDLE);
	hConHandle = _open_osfhandle(lStdHandle, _O_TEXT);
	fp = _fdopen( hConHandle, "w" );
	*stderr = *fp;
	setvbuf( stderr, NULL, _IONBF, 0 );
	// make cout, wcout, cin, wcin, wcerr, cerr, wclog and clog
	// point to console as well
	ios::sync_with_stdio();
}

void SetColorBackground(enum TEXTCOLORCODES cCode, HANDLE hstdout)
{
	switch (cCode){
	case 1:
		SetConsoleTextAttribute( hstdout, FOREGROUND_RED);
		return;
	case 2:
		SetConsoleTextAttribute( hstdout, FOREGROUND_RED |
			FOREGROUND_INTENSITY);
		return;
	case 3:
		SetConsoleTextAttribute( hstdout, FOREGROUND_GREEN);
		return;
	case 4:
		SetConsoleTextAttribute( hstdout, FOREGROUND_GREEN |
			FOREGROUND_INTENSITY);
		return;
	case 5:
		SetConsoleTextAttribute( hstdout, FOREGROUND_BLUE);
		return;
	case 6:
		SetConsoleTextAttribute( hstdout, FOREGROUND_BLUE |
			FOREGROUND_INTENSITY);
		return;
	case 7:
		SetConsoleTextAttribute( hstdout, FOREGROUND_RED |
			FOREGROUND_GREEN);
		return;
	case 8:
		SetConsoleTextAttribute( hstdout, FOREGROUND_RED |
			FOREGROUND_GREEN |
			FOREGROUND_INTENSITY);
		return;
	case 9:
		SetConsoleTextAttribute( hstdout, FOREGROUND_GREEN |
			FOREGROUND_BLUE);
		return;
	case 10:
		SetConsoleTextAttribute( hstdout, FOREGROUND_GREEN |
			FOREGROUND_BLUE |
			FOREGROUND_INTENSITY);
		return;
	case 11:
		SetConsoleTextAttribute( hstdout, FOREGROUND_BLUE |
			FOREGROUND_RED);
		return;
	case 12:
		SetConsoleTextAttribute( hstdout, FOREGROUND_BLUE |
			FOREGROUND_RED |
			FOREGROUND_INTENSITY);
		return;
	case 13:
		SetConsoleTextAttribute( hstdout, FOREGROUND_BLUE |
			FOREGROUND_RED |
			FOREGROUND_GREEN);
		return;
	case 14:
		SetConsoleTextAttribute( hstdout, FOREGROUND_BLUE |
			FOREGROUND_RED |
			FOREGROUND_GREEN |
			FOREGROUND_INTENSITY);
		return;
	default:
		return;
	}
}

void OutputConsoleText(char* sStrInfo, enum TEXTCOLORCODES iType)
{
	//We will need this handle to get the current background attribute
	HANDLE hStdOut = GetStdHandle(STD_OUTPUT_HANDLE);
	CONSOLE_SCREEN_BUFFER_INFO csbi;

	//We use csbi for the wAttributes word.
	if(GetConsoleScreenBufferInfo(hStdOut, &csbi))
	{
		//change color
		SetColorBackground(iType, hStdOut);
		printf("%s",sStrInfo);
		//restore color
		WORD wColor = (csbi.wAttributes & 0x0F);
		SetConsoleTextAttribute(hStdOut, wColor);
	}
	return;
}
//#endif
//End of File